package apis

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"strings"
	dto "vrcm/src/vrcm/app/admin/service/dto/live_schedule"
	service "vrcm/src/vrcm/app/admin/service/live_schedule"
	"vrcm/src/vrcm/common/apis"
)

type RegisterEquip struct {
	apis.Api
}

func (e RegisterEquip) GetPage(c *gin.Context) {
	req := dto.SourceBasicGetPageReq{}
	s := service.RegisterEquipService{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(apis.InternalError.Code, err, err.Error())
		return
	}

	var count int64
	data, err := s.GetEquipPage(&req, &count)
	if err != nil {
		e.Error(apis.DBQueryError.Code, err, fmt.Sprintf("获取注册失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.PageOK(data, int(count), req.GetPageIndex(), req.GetPageSize(), "查询成功")
}

func (e RegisterEquip) Get(c *gin.Context) {
	req := dto.SourceBasicGetReq{}
	s := service.SourceBasicService{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(apis.InternalError.Code, err, err.Error())
		return
	}
	data, err := s.Get(&req)
	if err != nil {
		e.Error(apis.DBQueryError.Code, err, fmt.Sprintf("获取信源失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(data, "查询成功")
}

func (e RegisterEquip) InsertEquip(c *gin.Context) {
	req := dto.SourceBasicInsertReq{}
	s := service.RegisterEquipService{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(apis.InternalError.Code, err, err.Error())
		return
	}

	err = s.InsertEquip(&req)
	if err != nil {
		if strings.Contains(err.Error(), "Duplicate entry") {
			e.Error(apis.RepeatError.Code, err, fmt.Sprintf("注册设备失败， %s", "设备已被注册"))
		} else {
			e.Error(apis.AddSourceError.Code, err, fmt.Sprintf("注册设备失败，%s", err.Error()))
		}
		return
	}

	e.OK(req.GetId(), "注册设备成功")
}

func (e RegisterEquip) UpdateEquip(c *gin.Context) {
	req := dto.EquipUpdateReq{}
	s := service.RegisterEquipService{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(apis.InternalError.Code, err, err.Error())
		return
	}

	err = s.UpdateEquip(&req)
	if err != nil {
		if strings.Contains(err.Error(), "Duplicate entry") {
			e.Error(apis.RepeatError.Code, err, fmt.Sprintf("修改设备失败， %s", "信源名不能重复"))
		} else if strings.Contains(err.Error(), "流地址不能重复") {
			e.Error(apis.SourceRepeatError.Code, err, fmt.Sprintf("修改设备失败， %s", "流地址不能重复"))
		} else {
			e.Error(apis.UpdateSourceError.Code, err, fmt.Sprintf("修改设备失败， %s", err.Error()))
		}

		return
	}
	e.OK(req.Id, "修改成功")
}

func (e RegisterEquip) DeleteEquip(c *gin.Context) {
	req := dto.SourceBasicDeleteReq{}
	s := service.SourceBasicService{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(apis.InternalError.Code, err, err.Error())
		return
	}

	err = s.Remove(&req)
	if err != nil {
		e.Error(apis.DeleteSourceGroupError.Code, err, fmt.Sprintf("删除设备失败， %s", err.Error()))
		return
	}
	e.OK(nil, "删除成功")
}

func (e RegisterEquip) GetByCode(c *gin.Context) {
	req := dto.GetEquipByCode{}
	s := service.RegisterEquipService{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(apis.InternalError.Code, err, err.Error())
		return
	}

	data, err := s.GetEquipByCode(&req)
	if err != nil {
		e.Error(apis.DBQueryError.Code, err, fmt.Sprintf("获取设备信息失败， %s", err.Error()))
		return
	}
	e.OK(data, "获取设备信息成功")
}

func (e RegisterEquip) EquipBeat(c *gin.Context) {
	s := service.SourceBasicService{}
	req := dto.EquipBeatReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(apis.InternalError.Code, err, err.Error())
		return
	}

	err = s.EquipBeat(&req)
	if err != nil {
		e.Error(apis.DBUpdateError.Code, err, fmt.Sprintf("设备状态回调失败， %s", err.Error()))
		return
	}
	e.OK("", "设备状态回调成功")
}
